from flask import Flask,request,render_template,redirect,url_for,flash
import 豆包AI文生图
import 可灵AI文生图
import 阿里云百炼AI文生图
import 即梦AI文生图
app = Flask(__name__)

@app.route('/')
def index():
    ais=[
        {'code':'doubao','name':'豆包AI'},
        {'code':'keling','name':'可灵AI'},
        {'code':'bailian','name':'百炼AI'},
        {'code':'jimeng','name':'即梦AI'}
    ]
    return render_template("login.html",ais=ais)
@app.route('/test/',methods=['GET','POST'])
def test():
    prompt=request.form['prompt']

    Art_style=request.form['Art_style']#选择画面风格
    prompt+='，风格为:'+str(Art_style)

    Size=request.form['Size']#选择画面尺寸
    prompt+='，图像比列为:'+str(Size)

    Shoot=request.form.get('Shoot')#选择拍摄方式
    prompt+='，拍摄方式:'+str(Shoot)

    Composition=request.form.get('Composition')#选择构图方式
    prompt+='，构图方式:'+str(Composition)

    Picture_effect=request.form.get('Picture_effect')#选择画面效果
    prompt+='，画面效果:'+str(Picture_effect)

    environmental=request.form.get('environmental')#选择环境氛围
    prompt+='，环境氛围:'+str(environmental)

    choice=request.form['choice']#选择的大模型
    print(choice,prompt)
    if choice=='doubao':
        image_url=豆包AI文生图.start(prompt)
        return render_template("photo.html",image_file=image_url+'.jpg')
    elif choice=='keling':
        image_url = 可灵AI文生图.start(prompt)
        return render_template("photo.html", image_file=image_url + '.jpg')
    elif choice=='bailian':
        image_url = 阿里云百炼AI文生图.start(prompt)
        return render_template("photo.html", image_file=image_url + '.jpg')
    elif choice=='jimeng':
        image_url = 即梦AI文生图.start(prompt)
        return render_template("photo.html", image_file=image_url + '.jpg')
    print('没返回')
    return '失败'
if __name__ == '__main__':
    app.run(host='0.0.0.0', port=5000, debug=True)
